import torch

BATCH_SIZE = 32
NUM_WORKERS = 4
RATE = 0.0001
EPOCHS = 15

train_data_root = ''
test_data_root = ''

TRAIN_DATA_PATH = r'J:/dog-vs-cat/train/train'
TEST_DATA_PATH = r'J:/dog-vs-cat/test/test'

MODEL_NAME = 'result.pkl'

IMAGE_WIDTH = 224
IMAGE_HEIGHT = 224

DEVICES = 'cuda:0' if torch.cuda.is_available() else 'cpu'